题目大意
寻找一组数排序的下一个序列
例如：1，2，3，下一个就是1，3，2
解题思路
官方思路（与下方相同）：https://leetcode-cn.com/problems/next-permutation/solution/
http://fisherlei.blogspot.com/2012/12/leetcode-next-permutation.html
图中文字翻译：
- 从右向左，找到第一个违反生序的，也就是图中的6，叫他‘partitionnumber’。
- 从右向左，找到第一个比刚才那个数大的数，叫他‘changenumber’。
- 交换着两个数，然后把在partitionnumber右边的数顺序反过来。
代码
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
| class Solution(object):
def nextPermutation(self, nums):
"""
:type nums: List[int]
:rtype: void Do not return anything, modify nums in-place instead.
"""
partition = -1
for i in range(len(nums)-2, -1, -1): # 从倒数第二个数开始遍历到第0个数
if nums[i] < nums[i+1]: # 从后向前找到第一个升序对，并让partition等于升序对中较小的
partition = i
break
if partition == -1:
# 没找到升序对 e.g. 54321 --> 12345
nums.reverse()
else:
for i in range(len(nums)-1, partition, -1):
# 交换partition和升序对后面比partition更大的数 如14532 -> 15432
if nums[i] > nums[partition]:
nums[i], nums[partition] = nums[partition], nums[i]
break
# 将partition后面的数字逆向排序。由于找到的partition是从后向前的第一个升序对，所以可以放心partition后面一定都是降序的
# 所以逆向排序后可以得到一个新的刚好大于之前permutation的next permutation
# 比如此时14532已经变为15432，则再将5（此时partition已经是5了）后面的432逆序排列得到234.则最终的数字变为15234
nums[partition+1:len(nums)] = nums[partition+1:len(nums)][::-1] # 切片
|
总结